package com.greate.community.service;

import com.greate.community.auth.wechatauth.domain.UserWeChat;
import com.greate.community.constant.CommunityConstant;
import com.greate.community.dao.UserMapper;
import com.greate.community.entity.User;
import com.greate.community.entity.dto.EmailLoginDTO;
import com.greate.community.jwt.JwtTokenUtil;
import com.greate.community.jwt.JwtUser;
import com.greate.community.util.*;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.concurrent.TimeUnit;


/**
 * 用户相关
 */
@Slf4j
@Service
public class UserService implements CommunityConstant {

    private final EmailUtil emailUtil;

    private final JwtTokenUtil jwtTokenUtil;

    private final UserMapper userMapper;

    private final RedisTemplate redisTemplate;


    public UserService(RedisTemplate redisTemplate,
                       EmailUtil emailUtil,
                       UserMapper userMapper,
                       JwtTokenUtil jwtTokenUtil) {
        this.redisTemplate = redisTemplate;
        this.emailUtil = emailUtil;
        this.userMapper = userMapper;
        this.jwtTokenUtil = jwtTokenUtil;
    }

    /**
     * 根据 Id 查询用户
     *
     * @param id
     * @return
     */
    public User findUserById(int id) {
        User user = getCache(id); // 优先从缓存中查询数据
        if (user == null) {
            user = initCache(id);
        }
        return user;
    }

    /**
     * 根据 Id 查询用户(包括实名信息)
     *
     * @param id
     * @return
     */
    public User findAllUserById(int id) {
        return userMapper.selectAllById(id);
    }

    /**
     * 根据 username 查询用户
     *
     * @param username
     * @return
     */
    public User findUserByName(String username) {
        return userMapper.selectByName(username);
    }

    /**
     * 根据 email 查询用户
     *
     * @param email
     * @return
     */
    public User findUserByEmail(String email) {
        return userMapper.selectByEmail(email);
    }

    /**
     * 用户注册
     *
     * @param user
     * @return Map<String, Object> 返回错误提示消息，如果返回的 map 为空，则说明注册成功
     */
    public Map<String, Object> register(User user) {
        Map<String, Object> map = new HashMap<>();

        // 验证账号是否已存在
        User u = userMapper.selectByName(user.getUsername());
        if (u != null) {
            map.put("userExistMsg", "该账号已存在");
            return map;
        }

        // 验证邮箱是否已存在
        u = userMapper.selectByEmail(user.getEmail());
        if (u != null) {
            map.put("emailExistMsg", "该邮箱已被注册");
            return map;
        }

        // 注册用户
        user.setSalt(CommunityUtil.generateUUID().substring(0, 5)); // salt
        user.setPassword(CommunityUtil.md5(user.getPassword() + user.getSalt())); // 加盐加密
        user.setType(0); // 默认普通用户
        user.setStatus(0); // 默认未激活

        // 给注册用户发送激活邮件
        String code = emailUtil.sendRegistrationCode(user.getEmail());
        user.setActivationCode(code); // 激活码

        // 随机头像（用户登录后可以自行修改）
        user.setHeaderUrl(String.format("http://images.nowcoder.com/head/%dt.png", new Random().nextInt(1000)));
        user.setCreateTime(new Date()); // 注册时间
        userMapper.insertUser(user);

        return map;
    }

    /**
     * 激活用户
     *
     * @param userId 用户 id
     * @param code   激活码
     * @return
     */
    public int activation(int userId, String code) {
        User user = userMapper.selectById(userId);
        if (user.getStatus() == 1) {
            // 用户已激活
            return ACTIVATION_REPEAT;
        } else if (user.getActivationCode().equals(code)) {
            // 修改用户状态为已激活
            userMapper.updateStatus(userId, 1);
            clearCache(userId); // 用户信息变更，清除缓存中的旧数据
            return ACTIVATION_SUCCESS;
        } else {
            return ACTIVATION_FAILURE;
        }
    }

    /**
     * 用户邮箱登录（为用户创建凭证）
     */
    public Map<String, Object> loginByEmail(EmailLoginDTO emailLoginDTO) {
        Map<String, Object> map = new HashMap<>();

        // 验证账号
        User user = userMapper.selectByEmail(emailLoginDTO.getEmail());
        if (user == null) {
            map.put("passwordMsg", "用户名或密码错误");
            return map;
        }

        // 验证状态
        if (user.getStatus() == 0) {
            // 账号未激活
            map.put("usernameMsg", "该账号未激活");
            return map;
        }

        // 验证密码
        String password = CommunityUtil.md5(emailLoginDTO.getPassword() + user.getSalt());
        if (!user.getPassword().equals(password)) {
            map.put("passwordMsg", "用户名或密码错误");
            return map;
        }

        // 生成token并返回, 此时的username用邮箱代替
        String token = jwtTokenUtil.generateToken(String.valueOf(user.getId()));

        JwtUser jwtUser = JwtUser.build(user);

        // 缓存token
        redisTemplate.opsForValue().set(RedisKeyUtil.getTokenKey(String.valueOf(user.getId())), jwtUser, jwtTokenUtil.getExpiration(), TimeUnit.SECONDS);

        map.put("userId", user.getId());
        map.put("token", token);

        return map;
    }

    public String loginByWechat(UserWeChat userWeChat) {

        String openid = userWeChat.getOpenId();
        String headerUrl = userWeChat.getHeadimgurl(); //微信头像地址

        // 若微信用户对应的账号已存在，跳过模拟注册操作，直接进入登录过程
        User user = userMapper.selectByOpenid(openid);
        if (user == null) {
            // 若不存在该微信用户，注册一个账号
            User newUser = new User();
            newUser.setType(0); // 默认普通用户
            newUser.setStatus(1); // 微信登录用户默认激活
            newUser.setLevel(0); // 等级默认0级
            newUser.setExp(0); // 经验0
            newUser.setLikeCount(0); // 点赞数量0

            // 用户头像
            newUser.setHeaderUrl(headerUrl); // 微信头像地址
            newUser.setCreateTime(new Date()); // 注册时间
            newUser.setOpenid(openid); // 微信用户openid

            userMapper.insertUser(newUser);
            user = userMapper.selectByOpenid(openid);
        }
        // 生成token并返回, 此时的username用邮箱代替
        String token = jwtTokenUtil.generateToken(String.valueOf(user.getId()));

        JwtUser jwtUser = JwtUser.build(user);

        // 缓存token
        redisTemplate.opsForValue().set(RedisKeyUtil.getTokenKey(String.valueOf(user.getId())), jwtUser, jwtTokenUtil.getExpiration(), TimeUnit.SECONDS);

        return token;
    }

    /**
     * 用户退出（删除缓存）
     *
     * @param userId
     */
    public void logout(String userId) {
        String redisKey = RedisKeyUtil.getTokenKey(userId);
        redisTemplate.delete(redisKey);
    }

    /**
     * 修改用户头像
     *
     * @param userId
     * @param headUrl
     * @return
     */
    public int updateHeader(int userId, String headUrl) {
        int rows = userMapper.updateHeader(userId, headUrl);
        clearCache(userId);
        return rows;
    }

    /**
     * 修改用户名
     *
     * @param userId
     * @param username
     * @return
     */
    public int updateUsername(int userId, String username) {
        User user = findUserByName(username);
        if (user == null) {
            // 名称没被占用
            int rows = userMapper.updateUsername(userId, username);
            clearCache(userId);
            return rows;
        } else if (user.getId() == userId) {
            // 新名称和原名称相同
            return 1;
        } else {
            // 名称被占用
            return 2;
        }
    }

    /**
     * 修改用户密码（对新密码加盐加密存入数据库）
     *
     * @param userId
     * @param newPassword 新密码
     * @return
     */
    public int updatePassword(int userId, String newPassword) {
        User user = userMapper.selectById(userId);
        // 重新加盐加密
        newPassword = CommunityUtil.md5(newPassword + user.getSalt());
        clearCache(userId);
        return userMapper.updatePassword(userId, newPassword);
    }

    /**
     * 优先从缓存中取值
     *
     * @param userId
     * @return
     */
    private User getCache(int userId) {
        String redisKey = RedisKeyUtil.getUserKey(userId);
        return (User) redisTemplate.opsForValue().get(redisKey);
    }

    /**
     * 缓存中没有该用户信息时，则将其存入缓存
     *
     * @param userId
     * @return
     */
    private User initCache(int userId) {
        User user = userMapper.selectById(userId);
        String redisKey = RedisKeyUtil.getUserKey(userId);
        redisTemplate.opsForValue().set(redisKey, user, 3600, TimeUnit.SECONDS);
        return user;
    }

    /**
     * 用户信息变更时清除对应缓存数据
     *
     * @param userId
     */
    private void clearCache(int userId) {
        String redisKey = RedisKeyUtil.getUserKey(userId);
        redisTemplate.delete(redisKey);
    }

    /**
     * 获取某个用户的权限
     *
     * @param userId
     * @return
     */
    public Collection<? extends GrantedAuthority> getAuthorities(int userId) {
        User user = this.findUserById(userId);
        List<GrantedAuthority> list = new ArrayList<>();
        list.add(new GrantedAuthority() {
            @Override
            public String getAuthority() {
                switch (user.getType()) {
                    case 1:
                        return AUTHORITY_ADMIN;
                    case 2:
                        return AUTHORITY_MANAGER;
                    case 3:
                        return AUTHORITY_HUSTER;
                    default:
                        return AUTHORITY_USER;
                }
            }
        });
        return list;
    }


    /**
     * 发送邮箱验证码
     *
     * @param email 账户邮箱
     * @return Map<String, Object> 返回错误提示消息，如果返回的 map 为空，则说明发送验证码成功
     */
    public Map<String, Object> doSendEmailCode4ResetPwd(String email) {
        Map<String, Object> map = new HashMap<>(2);
        User user = userMapper.selectByEmail(email);
        if (user == null) {
            map.put("errMsg", "该邮箱未绑定账号");
            return map;
        }

        // 给注册用户发送激活邮件
        String code = emailUtil.sendRetrievePasswordCode(email);

        final String redisKey = "EmailCode4ResetPwd:" + email;

        redisTemplate.opsForValue().set(redisKey, code, 600, TimeUnit.SECONDS);
        return map;
    }

    /**
     * 发送邮箱验证码
     *
     * @param email 账户邮箱
     * @return Map<String, Object> 返回错误提示消息，如果返回的 map 为空，则说明发送验证码成功
     */
    public Map<String, Object> doResetPwd(String email, String password) {
        Map<String, Object> map = new HashMap<>(2);
        if (StringUtils.isBlank(password)) {
            map.put("errMsg", "密码不能为空");
            return map;
        }
        User user = userMapper.selectByEmail(email);
        if (user == null) {
            map.put("errMsg", "未发现账号");
            return map;
        }
        final String passwordEncode = CommunityUtil.md5(password + user.getSalt());
        int i = userMapper.updatePassword(user.getId(), passwordEncode);
        if (i <= 0) {
            map.put("errMsg", "修改数据库密码错误");
        } else {
            clearCache(user.getId());
        }
        return map;
    }

    public int updateExpAndLevel(int userId, int exp, int level) {
        int rows = userMapper.updateExpAndLevel(userId, exp, level);
        clearCache(userId);
        return rows;
    }

    /**
     * 修改用户实名信息审核状态，保存用户实名信息，
     * 当审核通过时，同时修改用户身份为实名用户huster
     *
     * @param userId
     * @param auditStatus
     * @return
     */
    public int updateAuditStatus(int userId, int auditStatus) {
        int rows = userMapper.updateAuditStatus(userId, auditStatus);
        clearCache(userId);
        return rows;
    }

    /**
     * 分页查询实名信息（已激活的用户）
     *
     * @param auditStatus 当传入的 auditStatus = 0 时,查找所有 未上传学生证图片 的用户信息
     *                    当传入的 auditStatus = 1 时,查找所有 未审核 的用户信息
     *                    当传入的 auditStatus = 2 时,查找所有 审核通过 的用户信息
     *                    当传入的 auditStatus = 3 时,查找所有 审核驳回 的用户信息
     * @param offset      每页的起始索引
     * @param limit       每页显示多少条数据
     * @param detailed    是否查询实名详细信息，0-简略信息，1-详细信息
     * @return
     */
    public List<User> findRealNameList(int auditStatus, int offset, int limit, int detailed) {
        return userMapper.selectRealNameList(auditStatus, offset, limit, detailed);
    }

    /**
     * 查询实名用户个数
     *
     * @param auditStatus 当传入的 auditStatus = 0 时,查找所有 未上传学生证图片 的用户信息
     *                    当传入的 auditStatus = 1 时,查找所有 未审核 的用户信息
     *                    当传入的 auditStatus = 2 时,查找所有 审核通过 的用户信息
     *                    当传入的 auditStatus = 3 时,查找所有 审核驳回 的用户信息
     * @return
     */
    public int findRealNameRows(int auditStatus) {
        return userMapper.selectRealNameRows(auditStatus);
    }

    /**
     * 依据userId查询用户实名详细信息
     *
     * @param userId 用户id
     * @return
     */
    public User findRealNameUserById(int userId) {
        return userMapper.selectRealNameUserById(userId);
    }

    /**
     * 更新用户实名信息
     *
     * @param user
     * @return
     */
    public int updateRealNameInfo(User user) {
        int rows = userMapper.updateRealNameInfo(user);
        String redisKey = RedisKeyUtil.getRealNameKey();
        redisTemplate.opsForSet().add(redisKey, user.getId());
        return rows;
    }

    public int like(int id) {
        return userMapper.addLikeCount(id, 1);
    }

    public int cancelLike(int id) {
        return userMapper.addLikeCount(id, -1);
    }

}
